package leetcode.interviewClassics;

import java.util.Stack;

/**
 * @author Cheng Jun
 * Description: 请设计一个栈，除了常规栈支持的pop与push函数以外，还支持min函数，该函数返回栈元素中的最小值。
 * 执行push、pop和min操作的时间复杂度必须为O(1)。
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/min-stack-lcci
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 * @version 1.0
 * @date 2021/11/4 13:17\
 * @see MinStack1
 */
public class MinStack {

    // 因为使用了 辅助栈 minStack，所以空间复杂度为O(n)。如果想O(n)的空间复杂度降为O(1),可以看看 MinStack1
    private Stack<Integer> stack = new Stack<>();
    private Stack<Integer> minStack = new Stack<>();

    /**
     * initialize your data structure here.
     */
    public MinStack() {
    }

    public void push(int x) {
        // 入栈
        stack.push(x);
        // 入栈后，保存对应位置最小的 数
        if (minStack.isEmpty()) {
            minStack.push(x);
        } else {
            Integer peek = minStack.peek();
            if (peek < x) {
                minStack.push(peek);
            } else {
                minStack.push(x);
            }
        }
    }

    public void pop() {
        stack.pop();
        minStack.pop();
    }

    public int top() {

        return stack.peek();
    }

    public int getMin() {
        return minStack.peek();
    }
}
